import pandas as pd
import os

# 指定目录路径
directory = './成交查询_20240420152022.xlsx'

# # 读取目录下所有Excel文件
# files = [file for file in os.listdir(directory) if file.endswith('.xlsx')]
#
# # 创建一个空的DataFrame来存储所有数据
# df = pd.DataFrame()
#
# # 读取所有Excel文件并合并数据
# for file in files:
#
#     df = df.append(data)
data = pd.read_excel(directory)
# 按指定字段进行汇总统计
pivot_table = pd.pivot_table(data, index=["资金账号","证券代码","交易类型","订单来源"], values=['成交数量', '成交金额'], aggfunc={'成交数量': 'sum', '成交金额': 'sum'})

# 打印数据透视表
print(pivot_table)

# 保存数据透视表到本地
pivot_table.to_excel('./pivot_table.xlsx')
print
